#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from rwbench import ROOT
from os.path import join
from django.conf import settings

settings.configure(
  TEMPLATE_DIRS=(join(ROOT, "django"),),
  TEMPLATE_LOADERS=(
    (
      "django.template.loaders.cached.Loader",
      ("django.template.loaders.filesystem.Loader",),
    ),
  ),
)
from django.template import (
  loader as django_loader,
  Context as DjangoContext,
  Node,
  NodeList,
  Variable,
  TokenParser,
)
from django import template as django_template_module
from django.template import Library


# for django extensions.  We monkey patch our extensions in so that
# we don't have to initialize a more complex django setup.
django_extensions = django_template_module.Library()
django_template_module.builtins.append(django_extensions)


from rwbench import dateformat

django_extensions.filter(dateformat)


def var_or_none(x):
  if x is not None:
    return Variable(x)


# and more django extensions
@django_extensions.tag
def input_field(parser, token):
  p = TokenParser(token.contents)
  args = [p.value()]
  while p.more():
    args.append(p.value())
  return InputFieldNode(*args)


@django_extensions.tag
def textarea(parser, token):
  p = TokenParser(token.contents)
  args = [p.value()]
  while p.more():
    args.append(p.value())
  return TextareaNode(*args)


@django_extensions.tag
def form(parser, token):
  p = TokenParser(token.contents)
  args = []
  while p.more():
    args.append(p.value())
  body = parser.parse(("endform",))
  parser.delete_first_token()
  return FormNode(body, *args)


class InputFieldNode(Node):
  def __init__(self, name, type=None, value=None):
    self.name = var_or_none(name)
    self.type = var_or_none(type)
    self.value = var_or_none(value)

  def render(self, context):
    name = self.name.resolve(context)
    type = "text"
    value = ""
    if self.type is not None:
      type = self.type.resolve(context)
    if self.value is not None:
      value = self.value.resolve(context)
    tmpl = django_loader.get_template("_input_field.html")
    return tmpl.render(DjangoContext({"name": name, "type": type, "value": value}))


class TextareaNode(Node):
  def __init__(self, name, rows=None, cols=None, value=None):
    self.name = var_or_none(name)
    self.rows = var_or_none(rows)
    self.cols = var_or_none(cols)
    self.value = var_or_none(value)

  def render(self, context):
    name = self.name.resolve(context)
    rows = 10
    cols = 40
    value = ""
    if self.rows is not None:
      rows = int(self.rows.resolve(context))
    if self.cols is not None:
      cols = int(self.cols.resolve(context))
    if self.value is not None:
      value = self.value.resolve(context)
    tmpl = django_loader.get_template("_textarea.html")
    return tmpl.render(
      DjangoContext({"name": name, "rows": rows, "cols": cols, "value": value})
    )


class FormNode(Node):
  def __init__(self, body, action=None, method=None):
    self.body = body
    self.action = action
    self.method = method

  def render(self, context):
    body = self.body.render(context)
    action = ""
    method = "post"
    if self.action is not None:
      action = self.action.resolve(context)
    if self.method is not None:
      method = self.method.resolve(context)
    tmpl = django_loader.get_template("_form.html")
    return tmpl.render(
      DjangoContext({"body": body, "action": action, "method": method})
    )
